import numpy as np
import matplotlib.pyplot as plt
from scipy.stats import norm
import matplotlib

# 设置matplotlib支持中文显示
matplotlib.rcParams['font.family'] = 'SimHei'  # 指定默认字体为黑体
matplotlib.rcParams['axes.unicode_minus'] = False  # 解决负号'-'显示为方块的问题


def plot_normal_distribution(mean, std_dev):
    # 生成一组数据
    x = np.linspace(mean - 3 * std_dev, mean + 3 * std_dev, 1000)

    # 计算正态分布的概率密度函数
    y = norm.pdf(x, mean, std_dev)

    # 绘制正态分布曲线
    plt.plot(x, y, label=f'正态分布 (μ={mean}, σ={std_dev})')

    # 添加标题和标签
    plt.title('正态分布')
    plt.xlabel('X')
    plt.ylabel('概率密度')

    # 显示图例
    plt.legend()

    # 显示图形
    plt.show()


# 示例：绘制均值为0，标准差为1的正态分布
plot_normal_distribution(0, 1)